import torch
from torch import nn
from torch.nn import functional as  F

dropout = nn.Dropout(p=0.2)
x = torch.arange(10).float()
y = dropout(x)
print(x)
print(y)

# 我们只在训练的时候drop参数，以缩小模型，能够控制计算量，最重要的是，能够处理过拟合问题
# 但预测的时候，我们不会用dropout（只用配置eval参数，就能让模型避开dropout层）。